# demo02_shutil_os.py
# 导入两个模块
import shutil, os

# 练习：编写一个函数，将指定的文件拷贝到多个目录中
# copy_file_target('1.txt', [目录1, 目录2,...])
def copy_file_target(src_path, target_list):
    # 遍历列表获取每一个目标目录
    for item in target_list:
        # 如果目标目录不存在，需要先创建
        os.makedirs(item)
        # 拷贝文件
        shutil.copy(src_path, item)

# 练习：编写一个函数，将指定目录下所有扩展名为.txt文件重命名为.log
# rename_txt_to_log('/opt/mytest/hello')
def rename_txt_to_log(dir):
    # 获取目录中所有的文件
    files = os.listdir(dir)
    # print(files)
    # 得到每一个文件
    for item in files:
        # 获取文件的完整路径
        full_path = os.path.join(dir, item)
        # 判断文件后缀名是为.txt
        if os.path.isfile(full_path) and full_path.endswith('.txt'):
            # print(full_path)
            # 获取新的文件名称
            new_name = full_path[0:-4] + '.log'
            os.rename(full_path, new_name)
if __name__ == '__main__':
    # copy_file_target('/root/1.jpg', ['/opt/mytest/demo01', '/opt/mytest/demo02', '/opt/mytest/demo03'])
    rename_txt_to_log('/opt/mytest/hello')




